#include <stdio.h>
#include <stdlib.h>

// 恢复对称矩阵
void restoreMatrix(int *array, int n, int *matrix)
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j <= i; j++)
        {
            if (i == j)
            {
                matrix[i * n + j] = array[i * (i + 1) / 2 + j];
            }
            else
            {
                matrix[i * n + j] = array[i * (i + 1) / 2 + j];
                matrix[j * n + i] = array[i * (i + 1) / 2 + j];
            }
        }
    }
}

// 计算矩阵相加
void addMatrices(int *A, int *B, int *result, int n)
{
    for (int i = 0; i < n * n; i++)
    {
        result[i] = A[i] + B[i];
    }
}

// 计算矩阵相乘
void multiplyMatrices(int *A, int *B, int *result, int n)
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            result[i * n + j] = 0;
            for (int k = 0; k < n; k++)
            {
                result[i * n + j] += A[i * n + k] * B[k * n + j];
            }
        }
    }
}

int main()
{
    int n;
    printf("请输入矩阵的阶数n：");
    scanf("%d", &n);

    // 动态分配内存来存储下三角形元素数组
    int *inputArrayA = (int *)malloc((n * (n + 1) / 2) * sizeof(int));
    int *inputArrayB = (int *)malloc((n * (n + 1) / 2) * sizeof(int));

    printf("请输入矩阵A的下三角形元素：\n");
    for (int i = 0; i < n * (n + 1) / 2; i++)
    {
        scanf("%d", &inputArrayA[i]);
    }

    printf("请输入矩阵B的下三角形元素：\n");
    for (int i = 0; i < n * (n + 1) / 2; i++)
    {
        scanf("%d", &inputArrayB[i]);
    }

    // 动态分配内存来存储恢复的矩阵A和B
    int *matrixA = (int *)malloc(n * n * sizeof(int));
    int *matrixB = (int *)malloc(n * n * sizeof(int));

    // 恢复矩阵A和B
    restoreMatrix(inputArrayA, n, matrixA);
    restoreMatrix(inputArrayB, n, matrixB);

    // 计算A+B
    int *resultAdd = (int *)malloc(n * n * sizeof(int));
    addMatrices(matrixA, matrixB, resultAdd, n);

    // 计算A*B
    int *resultMultiply = (int *)malloc(n * n * sizeof(int));
    multiplyMatrices(matrixA, matrixB, resultMultiply, n);

    // 打印结果
    printf("A+B的结果为：\n");
    for (int i = 0; i < n * n; i++)
    {
        printf("%d ", resultAdd[i]);
        if ((i + 1) % n == 0)
        {
            printf("\n");
        }
    }

    printf("A*B的结果为：\n");
    for (int i = 0; i < n * n; i++)
    {
        printf("%d ", resultMultiply[i]);
        if ((i + 1) % n == 0)
        {
            printf("\n");
        }
    }

    // 释放动态分配的内存
    free(inputArrayA);
    free(inputArrayB);
    free(matrixA);
    free(matrixB);
    free(resultAdd);
    free(resultMultiply);

    return 0;
}